package com.examination.client.util;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;

/**
 * @author xiaodong
 * @Description:
 * @Date: 2021/10/20
 */
public class MD5Util {

    private static final Integer SALT_LENGTH = 12;

    /**
     * 16进制数字
     */
    private static final String HEX_NUMS_STR="0123456789abcdef";

    /**
     * 加密密码
     * @param password
     * @return
     * @throws Exception
     */
    public static String getEncryptedPwd(String password) throws Exception{
        try{
            //声明加密后的口令数组变量
            byte[] pwd = null;
            //随机数生成器
            SecureRandom random = new SecureRandom();
            //声明盐数组变量
            byte[] salt = new byte[SALT_LENGTH];
            //将随机数放入盐变量中
            random.nextBytes(salt);

            //获得加密的数据
            byte[] digest = encrypte(salt,password);

            //因为要在口令的字节数组中存放盐，所以加上盐的字节长度
            pwd = new byte[digest.length + SALT_LENGTH];
            //将盐的字节拷贝到生成的加密口令字节数组的前12个字节，以便在验证口令时取出盐
            System.arraycopy(salt, 0, pwd, 0, SALT_LENGTH);
            //将消息摘要拷贝到加密口令字节数组从第13个字节开始的字节
            System.arraycopy(digest, 0, pwd, SALT_LENGTH, digest.length);
            //将字节数组格式加密后的口令转化为16进制字符串格式的口令
            return byteToHexString(pwd);
        }catch(Exception e){
            throw new Exception("获取加密密码失败",e);
        }

    }

    /**
     * 验证密码是否正确
     * @param password
     * @param passwordInDb
     * @return
     * @throws Exception
     */
    public static boolean validPassword(String password, String passwordInDb) throws Exception {
        try{
            //将16进制字符串格式口令转换成字节数组
            byte[] pwdInDb = hexStringToByte(passwordInDb);
            //声明盐变量
            byte[] salt = new byte[SALT_LENGTH];
            //将盐从数据库中保存的口令字节数组中提取出来
            System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);

            //获得加密的数据
            byte[] digest = encrypte(salt,password);

            //声明一个保存数据库中口令消息摘要的变量
            byte[] digestInDb = new byte[pwdInDb.length - SALT_LENGTH];
            //取得数据库中口令的消息摘要
            System.arraycopy(pwdInDb, SALT_LENGTH, digestInDb, 0, digestInDb.length);
            //比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同
            if (Arrays.equals(digest, digestInDb)) {
                //口令正确返回口令匹配消息
                return true;
            } else {
                //口令不正确返回口令不匹配消息
                return false;
            }
        }catch(Exception e){
            throw new Exception("密码验证失败",e);
        }
    }

    /**
     * 将指定byte数组转换成16进制字符串（大写）
     * @return
     */
    public static String byteToHexString(byte[] bytes) {
        StringBuffer md5str = new StringBuffer();
        //把数组每一字节换成16进制连成md5字符串
        int digital;
        for (int i = 0; i < bytes.length; i++) {
            digital = bytes[i];
            if(digital < 0) {
                digital += 256;
            }
            if(digital < 16){
                md5str.append("0");
            }
            md5str.append(Integer.toHexString(digital));
        }
        return md5str.toString();
    }
    /**
     *
     * 根据盐生成密码
     * @param salt
     * @param passwrod
     * @return
     * @throws Exception
     */
    public static byte[] encrypte(byte[] salt,String passwrod) throws Exception{
        try{
            //声明消息摘要对象
            MessageDigest md = null;
            //创建消息摘要
            md = MessageDigest.getInstance("MD5");
            //将盐数据传入消息摘要对象
            md.update(salt);
            //将口令的数据传给消息摘要对象
            md.update(passwrod.getBytes("UTF-8"));
            //获得消息摘要的字节数组
            return md.digest();
        }catch(Exception e){
            throw new Exception("Md5解密失败",e);
        }
    }


    /**
     * 将16进制字符串转换成字节数组(大写)
     * @param hex
     * @return
     */
    public static byte[] hexStringToByte(String hex) {
        int len = (hex.length() / 2);
        byte[] result = new byte[len];
        char[] hexChars = hex.toCharArray();
        for (int i = 0; i < len; i++) {
            int pos = i * 2;
            result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4
                    | HEX_NUMS_STR.indexOf(hexChars[pos + 1]));
        }
        return result;
    }
    public static void main(String[] args) {
        String encryptedPwd=new String();
        try {
            encryptedPwd = getEncryptedPwd("123456sssWW");
            System.out.println(encryptedPwd);
        } catch (Exception e) {
            e.printStackTrace();
        }

        try {
            boolean b = validPassword("123456sssWW", encryptedPwd);
            System.out.println(b);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}

